import FileSaver from 'file-saver';
import * as XLSX from 'xlsx';

export default {
  // 导出Excel表格
  exportExcel(name, tableName) {
    // name表示生成excel的文件名 tableName表示表格的id
    var table = document.querySelector(tableName);
    if (!table) {
      console.error(tableName);
      return;
    }
    var new_table = table.cloneNode(true); // 克隆表格
    var rows = new_table.querySelectorAll('tr');
    rows.forEach(row => {
      for (var i = 0; i < 2; i++) {
        if (row.cells.length > 0) {
          row.deleteCell(0); // 删除行中的第一个单元格
        }
      }
    });
    // 删除偶数行
    var rows = new_table.querySelectorAll('tr');
    for (var i = rows.length - 1; i >= 0; i--) {
      if (i>=rows.length/2) { // 因为索引从0开始，所以这里检查的是 i % 2 === 1
        rows[i].parentNode.removeChild(rows[i]);
      }
    }

    // 将处理后的表格转换为工作簿
    var sel = XLSX.utils.table_to_book(new_table);
    var selIn = XLSX.write(sel, { bookType: 'xlsx', bookSST: true, type: 'array' });

    try {
      FileSaver.saveAs(new Blob([selIn], { type: 'application/octet-stream' }), name);
    } catch (e) {
      if (typeof console !== 'undefined') console.log(e, selIn);
    }
    return selIn;
  }
};
